# 在/ultralytics/file/yolo_weight路径，执行权重转换文件，保证转换后的文件保存在/ultralytics/file/yolo_weight
import os
import sys
# 添加yolo模块的路径
yolo_module_path = "/ultralytics/ultralytics"
if os.path.exists(yolo_module_path):
    sys.path.append(yolo_module_path)
from ultralytics import YOLO
###
# 1.使用绝对路径的权重路径
###
# 仪表盘的权重路径："/ultralytics/file/yolo_weight/dashboard-300-epochs-0808.pt"
# 球的权重路径："/ultralytics/file/yolo_weight/ball-seg-300-epochs-0805.pt"
model_path = "/ultralytics/file/yolo_weight/yolo11n-200.pt"
# Load a YOLO11n PyTorch model
model = YOLO(model_path)

# Export the model to TensorRT
model.export(format="engine",dynamic=False,half=True,imgsz=640,device=0)  # creates 'yolo11n.engine'
###
# 2.更改转换后的文件名称 
###

